Python技巧：在字符串列表每K个字符后插入特定值

来源: https://www.geeksforgeeks.org/python-给定字符串列表中每个-k-字母后插入值/

当处理字符串列表时，有时需要根据特定的模式或规则向列表中插入新的元素。本文将探讨如何在给定的字符串列表中，每隔K个字符后插入一个新的字符。这不仅是一个有趣的编程挑战，也是数据处理和文本分析中常见的需求。

方法一：利用枚举（enumerate）方法

# Python程序用于在给定字符串列表的每个K个字符后插入值
list1 = ['p', 'y', 't', 'h', 'o', 'n']
# 输出原始列表
print("Original list : " + str(list1))
# 设置插入字符k
k = 'G'
# 设置间隔N
N = 2
# 使用join()与enumerate()组合实现插入
output = list(''.join(i + k * (N % 2 == 1) for N, i in enumerate(list1)))
# 输出结果
print("The list after insertion : " + str(output))


输出结果：

Original list : ['p', 'y', 't', 'h', 'o', 'n']
The list after insertion : ['p', 'y', 'G', 't', 'h', 'G', 'o', 'n', 'G']


方法二：使用itertools模块

# Python程序用于在给定字符串列表的每个K个字符后插入值
from itertools import chain
list1 = ['p', 'y', 't', 'h', 'o', 'n']
# 输出原始列表
print("Original list : " + str(list1))
# 设置插入字符k
k = 'x'
# 设置间隔N
N = 3
# 使用itertools.chain组合实现插入
output = list(chain(*[list1[i : i + N] + [k] if len(list1[i : i + N]) == N else list1[i : i + N] for i in range(0, len(list1), N)]))
# 输出结果
print("The list after insertion : " + str(output))


输出结果：

Original list : ['p', 'y', 't', 'h', 'o', 'n']
The list after insertion : ['p', 'y', 't', 'x', 'h', 'o', 'n', 'x']